package com.halfsword.creativecraft.datagen;

import com.halfsword.creativecraft.CreativeCraft;
import com.halfsword.creativecraft.item.ItemLoader;
import net.minecraft.core.HolderLookup;
import net.minecraft.data.PackOutput;
import net.minecraft.data.tags.ItemTagsProvider;
import net.minecraft.tags.ItemTags;
import net.minecraft.world.level.block.Block;
import net.minecraftforge.common.data.ExistingFileHelper;
import org.jetbrains.annotations.Nullable;

import java.util.concurrent.CompletableFuture;

public class ItemTagGenerator extends ItemTagsProvider {
    public ItemTagGenerator(PackOutput p_275343_, CompletableFuture<HolderLookup.Provider> p_275729_, CompletableFuture<TagLookup<Block>> p_275322_
            , @Nullable ExistingFileHelper existingFileHelper) {
        super(p_275343_, p_275729_, p_275322_, CreativeCraft.MOD_ID, existingFileHelper);
    }

    @Override
    protected void addTags(HolderLookup.Provider provider) {
        this.tag(ItemTags.TRIMMABLE_ARMOR)
                .add(ItemLoader.HELMETSPACE.get())
                .add(ItemLoader.CHESTPLATESPACE.get())
                .add(ItemLoader.LEGGINGSSPACE.get())
                .add(ItemLoader.BOOTSSPACE.get())
                .add(ItemLoader.HELMETENDER.get())
                .add(ItemLoader.CHESTPLATEENDER.get())
                .add(ItemLoader.LEGGINGSENDER.get())
                .add(ItemLoader.BOOTSENDER.get())

        ;
        this.tag(ItemTags.MUSIC_DISCS)
                .add(ItemLoader.CORNFIELD_CHASE_DISC.get());
    }
}
